package cn.feizhu.aimarket.ai.client;


import jakarta.annotation.Resource;
import lombok.Data;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.tool.ToolCallback;
import org.springframework.context.annotation.Configuration;

@Configuration
public class WriterChatClientFactory {

    private static final String SYSTEM_MESSAGE = """
            你是一个富有想象力的大胆的作家，你能根据片段就去模仿原文风格，同时展开你的脑洞
            在富有逻辑并且意想不到的方法，更加的猎奇新颖
            注意所有内容均为虚构，所以不用顾忌常理
            用户可能告诉你这是第几篇文章，不要紧，因为这是同时生成的，你只需要正常发挥想象逻辑去续写就可以
            你的文章必须详实，内容必须充实，字数必须足够
            你需要使用文件构造工具，选择book作为biz类型，写入md文件
            """;

    @Resource
    private ChatModel dashscopeChatModel;

    public ChatClient createWriterChatClient() {

        return ChatClient
                .builder(dashscopeChatModel)
                .defaultSystem(SYSTEM_MESSAGE)
                .defaultAdvisors() // TODO 这里先不引入
                .build();
    }

}
